package com.tcd.explore.Util;

import com.jcraft.jsch.*;
import org.apache.commons.io.IOUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Value;

import java.io.*;
import java.util.Vector;

public class SftpUtil {
    private Logger logger = LogManager.getLogger(SftpUtil.class);

    private ChannelSftp sftp;

    private Session session;

    /**
     * 用户名
     */
    @Value("${ftp.userName}")
    private String username;

    /**
     * 密码
     */
    @Value("${ftp.password}")
    private String password;

    /**
     * 秘钥
     */
    private String privateKey;

    /**
     * FTP服务器Ip
     */
    @Value("${ftp.host}")
    private String host;

    /**
     * FTP服务器端口号
     */
    @Value("${ftp.port}")
    private int port;

    /**
     * 构造器：基于密码认证sftp对象
     * <p>
     * param1  用户名
     * param2  密码
     * param3   服务器ip
     * param4   服务器端口号
     */
    public SftpUtil(String username, String password, String host, int port) {
        this.username = username;
        this.password = password;
        this.host = host;
        this.port = port;
    }

    /**
     * 构造器：基于秘钥认证sftp对象
     *
     * @param username   用户名
     * @param privateKey 秘钥
     * @param host       服务器ip
     * @param port       服务器端口号
     */
    public SftpUtil(String username, String privateKey, int port, String host) {
        this.username = username;
        this.privateKey = privateKey;
        this.host = host;
        this.port = port;
    }

    /**
     * 连接SFTP服务器
     */
    public void login() {
        JSch jsch = new JSch();
        try {
            if (privateKey != null) {
                //设置登陆主机的秘钥
                jsch.addIdentity(privateKey);
            }

            //采用指定的端口连接服务器
            session = jsch.getSession(username, host, port);
            if (password != null) {
                //设置登陆主机的密码
                session.setPassword(password);
            }
            //优先使用 password 验证   注：session.connect()性能低，使用password验证可跳过gssapi认证，提升连接服务器速度
            session.setConfig("PreferredAuthentications", "password");
            //设置第一次登陆的时候提示，可选值：(ask | yes | no)
            session.setConfig("StrictHostKeyChecking", "no");
            session.connect();
            //创建sftp通信通道
            Channel channel = session.openChannel("sftp");
            channel.connect();
            sftp = (ChannelSftp) channel;
            logger.info("sftp server connect success !!");
        } catch (JSchException e) {
            logger.error("SFTP服务器连接异常！！", e);
        }
    }

    /**
     * 关闭SFTP连接
     */
    public void logout() {
        if (sftp != null) {
            if (sftp.isConnected()) {
                sftp.disconnect();
                logger.info("sftp is close already");
            }
        }
        if (session != null) {
            if (session.isConnected()) {
                session.disconnect();
                logger.info("session is close already");
            }
        }
    }

    /**
     * 将输入流上传到SFTP服务器，作为文件
     *
     * @param directory    上传到SFTP服务器的路径
     * @param sftpFileName 上传到SFTP服务器后的文件名
     * @param input        输入流
     * @throws SftpException
     */
    public void upload(String directory, String sftpFileName, InputStream input) throws SftpException {
        long start = System.currentTimeMillis();
        try {
            //如果文件夹不存在，则创建文件夹
            if (sftp.ls(directory) == null) {
                sftp.mkdir(directory);
            }
            //切换到指定文件夹
            sftp.cd(directory);
        } catch (SftpException e) {
            //创建不存在的文件夹，并切换到文件夹
            sftp.mkdir(directory);
            sftp.cd(directory);
        }
        sftp.put(input, sftpFileName);
        logger.info("文件上传成功！！ 耗时：{}ms", (System.currentTimeMillis() - start));
    }

    /**
     * 上传单个文件
     *
     * @param directory     上传到SFTP服务器的路径
     * @param uploadFileUrl 文件路径
     */
    public void upload(String directory, String uploadFileUrl) {
        File file = new File(uploadFileUrl);
        try {
            upload(directory, file.getName(), new FileInputStream(file));
        } catch (FileNotFoundException | SftpException e) {
            logger.error("上传文件异常！", e);
        }
    }

    /**
     * 将byte[] 上传到SFTP服务器，作为文件
     * 注： 从String转换成byte[] 需要指定字符集
     *
     * @param directory    上传到SFTP服务器的路径
     * @param sftpFileName 上传SFTP服务器后的文件名
     * @param bytes        字节数组
     */
    public void upload(String directory, String sftpFileName, byte[] bytes) {
        try {
            upload(directory, sftpFileName, new ByteArrayInputStream(bytes));
        } catch (SftpException e) {
            logger.error("上传文件异常！", e);
        }
    }

    /**
     * 将字符串按照指定编码格式上传到SFTP服务器
     *
     * @param directory    上传到SFTP服务器的路径
     * @param sftpFileName 上传SFTP服务器后的文件名
     * @param dataStr      字符串
     * @param charsetName  字符串的编码格式
     */
    public void upload(String directory, String sftpFileName, String dataStr, String charsetName) {
        try {
            upload(directory, sftpFileName, new ByteArrayInputStream(dataStr.getBytes(charsetName)));
        } catch (UnsupportedEncodingException | SftpException e) {
            logger.error("上传文件异常！", e);
        }
    }

    /**
     * 下载文件
     *
     * @param directory    SFTP服务器的文件路径
     * @param downloadFile SFTP服务器上的文件名
     * @param saveFile     保存到本地路径
     */
    public void download(String directory, String downloadFile, String saveFile) {
        try {
            if (directory != null && !"".equals(directory)) {
                sftp.cd(directory);
            }
            File file = new File(saveFile);
            sftp.get(downloadFile, new FileOutputStream(file));
        } catch (SftpException | FileNotFoundException e) {
            logger.error("文件下载异常！", e);
        }
    }

    /**
     * 下载文件
     *
     * @param directory    SFTP服务器的文件路径
     * @param downloadFile SFTP服务器上的文件名
     * @return 字节数组
     */
    public byte[] download(String directory, String downloadFile) {
        try {
            if (directory != null && !"".equals(directory)) {
                sftp.cd(directory);
            }
            InputStream inputStream = sftp.get(downloadFile);
            return IOUtils.toByteArray(inputStream);
        } catch (SftpException | IOException e) {
            logger.error("文件下载异常！", e);
        }
        return null;
    }

    /**
     * 下载文件
     *
     * @param directory    SFTP服务器的文件路径
     * @param downloadFile SFTP服务器上的文件名
     * @return 输入流
     */
    public InputStream downloadStream(String directory, String downloadFile) {
        try {
            if (directory != null && !"".equals(directory)) {
                System.out.println("文件地址為" + directory);
                sftp.cd(directory);
            }
            InputStream input = sftp.get(downloadFile);
            return input;
        } catch (SftpException e) {
            logger.error("文件下载异常！", e);
        }
        return null;
    }

    /**
     * 删除文件
     *
     * @param directory      SFTP服务器的文件路径
     * @param deleteFileName 删除的文件名称
     */
    public void delete(String directory, String deleteFileName) {
        try {
            sftp.cd(directory);
            sftp.rm(deleteFileName);
        } catch (SftpException e) {
            logger.error("文件删除异常！", e);
        }
    }

    /**
     * 删除文件夹
     *
     * @param directory SFTP服务器的文件路径
     */
    public void delete(String directory) {
        Vector vector = listFiles(directory);
        vector.remove(0);
        vector.remove(0);
        for (Object v : vector) {
            ChannelSftp.LsEntry lsEntry = (ChannelSftp.LsEntry) v;
            try {
                sftp.cd(directory);
                sftp.rm(lsEntry.getFilename());
            } catch (SftpException e) {
                logger.error("文件删除异常！", e);
            }
        }
    }

    /**
     * 获取文件夹下的文件
     *
     * @param directory 路径
     * @return
     */
    public Vector<?> listFiles(String directory) {
        try {
            if (isDirExist(directory)) {
                Vector<?> vector = sftp.ls(directory);
                //移除上级目录和根目录："." ".."
                vector.remove(0);
                vector.remove(0);
                return vector;
            }
        } catch (SftpException e) {
            logger.error("获取文件夹信息异常！", e);
        }
        return null;
    }

    /**
     * 检测文件夹是否存在
     *
     * @param directory 路径
     * @return
     */
    public boolean booleanUrl(String directory) {
        try {
            if (sftp.ls(directory) == null) {
                return false;
            }
        } catch (Exception e) {
            logger.error("检测文件夹异常！", e);
        }
        return true;
    }

    /**
     * 创建一个文件目录
     *
     * @param createpath 路径
     * @return
     */
    public boolean createDir(String createpath) {
        try {
            if (isDirExist(createpath)) {
                this.sftp.cd(createpath);
                return true;
            }
            String pathArry[] = createpath.split("/");
            StringBuffer filePath = new StringBuffer("/");
            for (String path : pathArry) {
                if (path.equals("")) {
                    continue;
                }
                filePath.append(path + "/");
                if (isDirExist(filePath.toString())) {
                    sftp.cd(filePath.toString());
                } else {
                    // 建立目录
                    sftp.mkdir(filePath.toString());
                    // 进入并设置为当前目录
                    sftp.cd(filePath.toString());
                }
            }
            this.sftp.cd(createpath);
        } catch (SftpException e) {
            logger.error("目录创建异常！", e);
            return false;
        }
        return true;
    }

    /**
     * 判断目录是否存在
     *
     * @param directory 路径
     * @return
     */
    public boolean isDirExist(String directory) {
        boolean isDirExistFlag = false;
        try {
            SftpATTRS sftpATTRS = this.sftp.lstat(directory);
            isDirExistFlag = true;
            return sftpATTRS.isDir();
        } catch (Exception e) {
            if (e.getMessage().toLowerCase().equals("no such file")) {
                isDirExistFlag = false;
            }
        }
        return isDirExistFlag;
    }

    /**
     * 方法功能说明：目录不存在时创建目录
     *
     * @return void
     * @throws
     * @参数： @param path
     */
    public void mkdirs(String path) {
        File file = new File(path);
        String fs = file.getParent();
        file = new File(fs);
        if (!file.exists()) {
            file.mkdirs();
        }
    }

    /**
     * 判断文件或目录是否存在
     *
     * @param
     * @return
     */
    public boolean isExist(String path, ChannelSftp sftp) {
        boolean isExist = false;
        try {
            sftp.lstat(path);
            isExist = true;
        } catch (Exception e) {
            if (e.getMessage().toLowerCase().equals("no such file")) {
                isExist = false;
            }
        }
        return isExist;
    }
}
